package com.ruoyi.framework.utils;

import cn.hutool.core.net.NetUtil;
import com.alibaba.fastjson.JSONObject;
import net.dreamlu.mica.http.HttpRequest;
import net.dreamlu.mica.ip2region.core.Ip2regionSearcher;
import net.dreamlu.mica.ip2region.core.IpInfo;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.context.annotation.Configuration;
import org.springframework.util.StringUtils;

/**
 * 获取地址类
 *
 * @author ruoyi
 */
@Configuration
public class AddressUtils {
    // IP地址查询
    public static final String IP_URL = "http://whois.pconline.com.cn/ipJson.jsp";
    // 未知地址
    public static final String UNKNOWN = "XX XX";
    private static final Logger log = LoggerFactory.getLogger(AddressUtils.class);
    private static Ip2regionSearcher regionSearcher;

    public AddressUtils(Ip2regionSearcher regionSearcher) {
        AddressUtils.regionSearcher = regionSearcher;
    }

    public static String getRealAddressByIP(String ip, Boolean isRemote) {
        String address = UNKNOWN;
        // 内网不查询
        if (NetUtil.isInnerIP(ip)) {
            return "内网IP";
        }
        try {
            String rspStr = HttpRequest.get(IP_URL)
                    .query("ip", ip)
                    .query("json", true).execute().asString();
            if (StringUtils.isEmpty(rspStr)) {
                log.error("获取地理位置异常 {}", ip);
                return UNKNOWN;
            }
            JSONObject obj = JSONObject.parseObject(rspStr);
            String region = obj.getString("pro");
            String city = obj.getString("city");
            return String.format("%s %s", region, city);
        } catch (Exception e) {
            log.error("获取地理位置异常 {}", ip);
        }
        return address;
    }

    public static String getRealAddressByIP(String ip) {
        String address = UNKNOWN;
        // 内网不查询
        try {
            IpInfo ipInfo = regionSearcher.memorySearch(ip);
            if (ipInfo != null) {
                return String.format("%s %s", ipInfo.getProvince(), ipInfo.getCity());
            }
        } catch (Exception e) {
            log.error("获取地理位置异常 {}", ip);
        }
        return address;
    }
}
